Goto

Collaborating Authors

 full stack machine learning project


A Full Stack Machine Learning Project

#artificialintelligence

When I was a kid, I used to be obsessed with Harry Potter (the books, not the movie). I was a huge fan of Hermione Granger, and was deeply intrigued by the character of Severus Snape. So when I found an MBTI personality prediction dataset, I decided that there was no better way to use it than create a Harry Potter character prediction model. As you can see, the interface takes in some text as input, calls the back-end model, and generates a prediction. As (aspiring) data scientists, our focus is mainly on the data and building models. Most of our machine learning projects lie in a carefully formatted Jupyter Notebook, and will probably stay there forever.